find sum of diagonal elements in python